package com.godfrey.springcloud.controller;

import com.godfrey.springcloud.pojo.Dept;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.client.RestTemplate;

import java.util.List;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * @author godfrey
 * @since 2020-11-18
 */
@RestController
public class DeptConsumerController {

    private RestTemplate restTemplate;

    private static final String REST_URL_PREFIX = "http://localhost:8001";

    private AtomicInteger num = new AtomicInteger(3);

    @Autowired
    DeptConsumerController(RestTemplate restTemplate) {
        this.restTemplate = restTemplate;
    }

    @PostMapping("/consumer/dept/add")
    public boolean addDept(Dept dept) {
        return restTemplate.postForObject(REST_URL_PREFIX + (num.getAndIncrement() % 3 + 1) + "/dept/add/", dept, Boolean.class);
    }

    @RequestMapping("/consumer/dept/get/{id}")
    public Dept queryById(@PathVariable("id") Long id) {
        return restTemplate.getForObject(REST_URL_PREFIX + (num.getAndIncrement() % 3 + 1) + "/dept/get/" + id, Dept.class);
    }

    @RequestMapping("/consumer/dept/list")
    public List<Dept> queryAll() {
        return restTemplate.getForObject(REST_URL_PREFIX + (num.getAndIncrement() % 3 + 1) + "/dept/list", List.class);
    }
}
